# 数据库地址
HOST = "127.0.0.1"

# 数据库端口
PORT = 3306

# 用户名
USER = "root"

# 用户密码
PASSWORD = "mysql"

# 数据库名字
DBNAME = "test"

# 数据库编码
CHARSET = "utf8"


# 使用with简化连接过程，每次都连接关闭很麻烦，使用上下文管理，简化连接过程
import pymysql
import contextlib


# 定义上下文管理器，连接后自动关闭连接
@contextlib.contextmanager
def mysql():
    conn = pymysql.connect(host=HOST, port=PORT, user=USER, password=PASSWORD, database=DBNAME, charset=CHARSET)
    cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
    try:
        yield cursor
    finally:
        conn.commit()
        cursor.close()
        conn.close()